Split JSON file into multiple files using Python(notebook)